CUDA_VISIBLE_DEVICES=0 python run_mlm_no_trainer.py \
    --train_file train.txt \
    --model_name_or_path chinese-roberta-wwm-ext \
    --line_by_line True \
    --output_dir test-mlm